iT邦幫忙

2018 iT 邦幫忙鐵人賽
DAY 14
0

這篇主要就是用來建制之後android會用到的模組,首先我們要建立的是下圖的這個感知機
imgur
下方的code就是我們用來創建模組以及把他的圖像和變數儲存起來

import tensorflow as tf

I = tf.placeholder(tf.float32,shape=[None,3],name='I')
W = tf.Variable(tf.zeros(shape=[3,2]),dtype=tf.float32,name='W')
b = tf.Variable(tf.zeros(shape=[2]),dtype=tf.float32,name='b')
O = tf.nn.relu(tf.matmul(I,W)+b,name='O')

saver = tf.train.Saver()
init_op = tf.global_variables_initializer()

with tf.Session() as sess:
  sess.run(init_op)
  tf.train.write_graph(sess.graph_def, '.', 'android_tf.pbtxt')  
  sess.run(tf.assign(W,[[1,2],[3,4],[5,6]]))
  sess.run(tf.assign(b,[1,2]))
  saver.save(sess, 'android_tf.ckpt')

上一篇
[Day 13] 介紹梯度下降法
下一篇
[Day 15] Freezing Graph
系列文
如何使用tensorflow在android上30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言